package handler

import (
	"context"
	"log"

	"google.golang.org/grpc/stats"
)

/**
* @Description: 自定义handler实现stats.Handler打印一些信息。由archive/server/main.go，根据需要引用
* @Author: liangan
* @Date: 2023-09-02_10:02
*/
type MyHandler struct {
}

func (h *MyHandler) TagRPC(c context.Context, tag *stats.RPCTagInfo) context.Context {
	log.Printf("TagRPC: %v", tag)
	return c
}

func (h *MyHandler) HandleRPC(c context.Context, s stats.RPCStats) {
	log.Printf("HandleRPC: %v", s)
}

func (h *MyHandler) TagConn(c context.Context, tag *stats.ConnTagInfo) context.Context {
	log.Printf("TagConn: %v", tag)
	return c
}
func (h *MyHandler) HandleConn(c context.Context, s stats.ConnStats) {
	log.Printf("HandleConn: %v", s)
}
